How to convert raw image bytes read from stream to a tensor of a shape valid to perform Conv2d